package cn.cxq.learning.tree;

/**
 * 剑指offer
 * 二叉树的下一个结点 相似的企业真题
 * 时间限制：C/C++ 1秒，其他语言2秒 空间限制：C/C++ 64M，其他语言128M 热度指数：380679
 * 本题知识点： 树
 *  算法知识视频讲解
 * 题目描述
 * 给定一个二叉树其中的一个结点，请找出中序遍历顺序的下一个结点并且返回。注意，树中的结点不仅包含左右子结点，同时包含指向父结点的next指针。
 * 说明：本题目包含复杂数据结构TreeLinkNode，点此查看相关信息
 */
public class GetNext {

    // 思路：先考虑其右节点是否为空，没有的话就遍历右子树找到下一个节点
    // 如果没有右节点，那么就找他是哪个节点（当然是第一个找到的）的左子树的一部分，然后这个节点就是他的下一个节点
    public TreeLinkNode GetNext(TreeLinkNode pNode) {

        if (pNode == null) {
            return null;
        }

        TreeLinkNode next;

        if (pNode.right!=null) {
            next = pNode.right;
            while (next.left!=null) {
                next = next.left;
            }
            return next;
        }

        if (pNode.next!=null) {
            next = pNode;

            while (next.next!=null && next.next.right == next) {
                next = next.next;
            }

            next = next.next;

            return next;
        }

        return null;
    }
}

class TreeLinkNode {
    int val;
    TreeLinkNode left = null;
    TreeLinkNode right = null;
    TreeLinkNode next = null;

    TreeLinkNode(int val) {
        this.val = val;
    }
}